package com.didispace.chapter11.utils;

import lombok.extern.slf4j.Slf4j;
import org.apache.commons.net.ftp.FTPClient;
import org.apache.commons.net.ftp.FTPFile;
import org.apache.commons.net.ftp.FTPReply;

import java.io.File;
import java.io.IOException;
import java.net.SocketException;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * @Author XIAOFENG
 * @Description //TODO
 * @Date Creat in 15:49 2021/6/7
 * @Modified By:
 */
@Slf4j
public class DeviceUtil {


    public static void getAllFileName(FTPClient ftp, String path, List<Map<String, String>> directoryName) {
//    public static void getAllFileName(String path, List<Map<String, String>> directoryName) {
//        boolean flag = false;
//        File file = new File(path);
//        File[] tempList = file.listFiles();
//
//        for (int i = 0; i < tempList.length; i++) {
//            Map<String, String> map = new HashMap<>();
//            if (tempList[i].isDirectory()) {
//
//                if (tempList[i].list().length > 0) {
//                    map.put(tempList[i].getName(), "1");
//                } else {
//                    map.put(tempList[i].getName(), "0");
//                }
//                directoryName.add(map);
//                getAllFileName(tempList[i].getAbsolutePath(), directoryName);
//            }
//        }
//        return;
        try {
            ftp.changeWorkingDirectory(new String(path.getBytes("GBK"),"ISO-8859-1"));
            //设置FTP连接模式
            ftp.enterLocalPassiveMode();
            //获取指定目录下文件文件对象集合
            FTPFile tempList[] = ftp.listFiles();

            for (int i = 0; i < tempList.length; i++) {
                Map<String, String> map = new HashMap<>();
                if (tempList[i].isDirectory()) {
                    String cpath = path +"/"+ tempList[i].getName();
                    ftp.changeWorkingDirectory(new String(cpath.getBytes("GBK"),"ISO-8859-1"));
                    //设置FTP连接模式
                    ftp.enterLocalPassiveMode();
                    //获取指定目录下文件文件对象集合
                    FTPFile childDirectory[] = ftp.listFiles();
                    if (childDirectory.length > 0) {
                        map.put(tempList[i].getName(), "1");
                    } else {
                        map.put(tempList[i].getName(), "0");
                    }
                    directoryName.add(map);
                }
            }
        } catch (Exception e) {
            e.printStackTrace();
        }
    }


    public static boolean makeDerectory(FTPClient ftp,String Path){
        boolean flag = false;
        try {
            flag = ftp.changeWorkingDirectory(new String(Path.getBytes("GBK"),"ISO-8859-1"));
            if(!ftp.changeWorkingDirectory(new String(Path.getBytes("GBK"),"ISO-8859-1"))){
                ftp.makeDirectory(new String(Path.getBytes("GBK"),"ISO-8859-1"));
            }
        } catch (IOException e) {
            e.printStackTrace();
        }
        return flag;
    }

    /**
     * 获取FTPClient对象
     * @param ftpHost 服务器IP
     * @param ftpPort 服务器端口号
     * @param ftpUserName 用户名
     * @param ftpPassword 密码
     * @return FTPClient
     */
    public  static FTPClient getFTPClient(String ftpHost, int ftpPort,
                                  String ftpUserName, String ftpPassword) {

        FTPClient ftp = null;
        try {
            ftp = new FTPClient();
            // 连接FPT服务器,设置IP及端口
            ftp.connect(ftpHost, ftpPort);
            // 设置用户名和密码
            ftp.login(ftpUserName, ftpPassword);
            // 设置连接超时时间,5000毫秒
            ftp.setConnectTimeout(50000);
            // 设置中文编码集，防止中文乱码
            ftp.setControlEncoding("UTF-8");
            if (!FTPReply.isPositiveCompletion(ftp.getReplyCode())) {
                log.info("未连接到FTP，用户名或密码错误");
                ftp.disconnect();
            } else {
                log.info("FTP连接成功");
            }

        } catch (SocketException e) {
            e.printStackTrace();
            log.info("FTP的IP地址可能错误，请正确配置");
        } catch (IOException e) {
            e.printStackTrace();
            log.info("FTP的端口错误,请正确配置");
        }
        return ftp;
    }

    /**
     * 关闭FTP方法
     * @param ftp
     * @return
     */
    public static boolean closeFTP(FTPClient ftp){

        try {
            ftp.logout();
            log.info("FTP关闭成功");
        } catch (Exception e) {
            log.error("FTP关闭失败");
        }finally{
            if (ftp.isConnected()) {
                try {
                    ftp.disconnect();
                } catch (IOException ioe) {
                    log.error("FTP关闭失败");
                }
            }
        }

        return false;

    }

    /**
     * 实现文件的移动，这里做的是一个文件夹下的所有内容移动到新的文件，
     * 如果要做指定文件移动，加个判断判断文件名
     * 如果不需要移动，只是需要文件重命名，可以使用ftp.rename(oleName,newName)
     * @param ftp
     * @param oldPath
     * @param newPath
     * @return
     */
    public static boolean moveFile(FTPClient ftp,String oldPath,String newPath){
        boolean flag = false;

        try {
            ftp.changeWorkingDirectory(new String(oldPath.getBytes("GBK"),"ISO-8859-1"));
            ftp.enterLocalPassiveMode();
            //获取文件数组
            FTPFile[] files = ftp.listFiles();
            //新文件夹不存在则创建
            if(!ftp.changeWorkingDirectory(newPath)){
                ftp.makeDirectory(newPath);
            }
            //回到原有工作目录
            ftp.changeWorkingDirectory(new String(oldPath.getBytes("GBK"),"ISO-8859-1"));

            for (int i = 0; i < files.length; i++) {
                if (i == 0) {
                    //转存目录
                    flag = ftp.rename(new String(files[i].getName().getBytes("GBK"),"ISO-8859-1"), newPath+File.separator+new String(files[i].getName().getBytes("GBK"),"ISO-8859-1"));
                    if(flag){
                        log.info(files[i].getName()+"移动成功");
                        System.out.println(files[i].getName()+"移动成功");
                    }else{
                        log.error(files[i].getName()+"移动失败");
                        System.out.println(files[i].getName()+"移动失败");
                    }
                }
            }

//            for (FTPFile file : files) {
//
//                //转存目录
//                flag = ftp.rename(new String(file.getName().getBytes("UTF-8"),"ISO-8859-1"), newPath+File.separator+new String(file.getName().getBytes("UTF-8"),"ISO-8859-1"));
//                if(flag){
//                    logger.info(file.getName()+"移动成功");
//                }else{
//                    logger.error(file.getName()+"移动失败");
//                }
//            }
        } catch (Exception e) {
            e.printStackTrace();
            log.error("移动文件失败");
        }
        return flag;
    }

    /**
     * 删除FTP上指定文件夹下文件及其子文件方法，添加了对中文目录的支持
     * @param ftp FTPClient对象
     * @param FtpFolder 需要删除的文件夹
     * @return
     */
    public static boolean deleteByFolder(FTPClient ftp,String FtpFolder){
        boolean flag = false;
        try {
//            ftp.changeWorkingDirectory(new String(FtpFolder.getBytes("UTF-8"),"ISO-8859-1"));
            ftp.changeWorkingDirectory(new String(FtpFolder.getBytes("GBK"),"ISO-8859-1"));
            ftp.enterLocalPassiveMode();
            FTPFile[] files = ftp.listFiles();
            for (FTPFile file : files) {
                //判断为文件则删除
                if(file.isFile()){
//                    ftp.deleteFile(new String(file.getName().getBytes("UTF-8"),"ISO-8859-1"));
                    boolean deleteFile = ftp.deleteFile(new String(file.getName().getBytes("GBK"), "ISO-8859-1"));
                    if(deleteFile){
                        log.info( "文件删除成功,fileName:{}",file.getName());
                    }else{
                        log.error( "文件删除失败,fileName:{}",file.getName());
                    }
                }
                //判断是文件夹
                if(file.isDirectory()){
                    String childPath = FtpFolder + File.separator+file.getName();
                    //递归删除子文件夹
                    deleteByFolder(ftp,childPath);
                }
            }
            //循环完成后删除文件夹
//            flag = ftp.removeDirectory(new String(FtpFolder.getBytes("UTF-8"),"ISO-8859-1"));
            flag = ftp.removeDirectory(new String(FtpFolder.getBytes("GBK"),"ISO-8859-1"));
            if(flag){
                log.info(FtpFolder+"文件夹删除成功");
            }else{
                log.error(FtpFolder+"文件夹删除失败");
            }

        } catch (Exception e) {
            e.printStackTrace();
            log.error("删除失败");
        }
        return flag;

    }




    /**
     * 删除文件，可以是文件或文件夹
     *
     * @param fileName：要删除的文件名
     * @return 删除成功返回true，否则返回false
     */
    public static boolean delete(String fileName) {
        File file = new File(fileName);
        if (!file.exists()) {
//            System.out.println("删除文件失败:" + fileName + "不存在！");
            log.info("删除文件失败:" + fileName + "不存在！");
            return false;
        } else {
            if (file.isFile())
                return deleteFile(fileName);
            else
                return deleteDirectory(fileName);
        }
    }

    /**
     * 删除单个文件
     *
     * @param filePath：要删除的文件的路径
     * @return 单个文件删除成功返回true，否则返回false
     */
    public static boolean deleteFile(String filePath) {
        File file = new File(filePath);
        // 如果文件路径所对应的文件存在，并且是一个文件，则直接删除
        if (file.exists() && file.isFile()) {
            if (file.delete()) {
//                System.out.println("删除单个文件" + file.getName() + "成功！");
                return true;
            } else {
//                System.out.println("删除单个文件" + file.getName() + "失败！");
                log.info("删除单个文件" + file.getName() + "失败！");
                return false;
            }
        } else {
//            System.out.println("删除单个文件失败：" + file.getName() + "不存在！");
            log.info("删除单个文件失败：" + file.getName() + "不存在！");
            return false;
        }
    }

    /**
     * 删除目录及目录下的文件
     *
     * @param dir：要删除的目录的文件路径
     * @return 目录删除成功返回true，否则返回false
     */
    public static boolean deleteDirectory(String dir) {
        // 如果dir不以文件分隔符结尾，自动添加文件分隔符
        if (!dir.endsWith(File.separator))
            dir = dir + File.separator;
        File dirFile = new File(dir);


        // 如果dir对应的文件不存在，或者不是一个目录，则退出
        if ((!dirFile.exists()) || (!dirFile.isDirectory())) {
//            System.out.println("删除目录失败：" + dir + "不存在！");
            log.info("删除目录失败：" + dir + "不存在！");
            return false;
        }
        boolean flag = true;
        // 删除文件夹中的所有文件包括子目录
        File[] files = dirFile.listFiles();
        for (int i = 0; i < files.length; i++) {

            // 删除子文件
            if (files[i].isFile()) {
                flag = deleteFile(files[i].getAbsolutePath());
                if (!flag)
                    break;
            }
            // 删除子目录
            else if (files[i].isDirectory()) {
                flag = deleteDirectory(files[i].getAbsolutePath());
                if (!flag)
                    break;
            }
        }
        if (!flag) {
//            System.out.println("删除目录失败！");
            log.info("删除目录失败！");
            return false;
        }
        // 删除当前目录
        if (dirFile.delete()) {
            return true;
        } else {
            return false;
        }
    }

    public static void move(String path,String copyDirectory) {
        File file = new File(path);
        File[] files = file.listFiles();

        for (int i = 0; i < files.length; i++) {
            if (i == 0) {
                File copyFile = new File(copyDirectory);
                if (!copyFile.exists() && !copyFile.isDirectory()) {
                    System.out.println("//不存在");
                    copyFile.mkdirs();
                } else {
                    System.out.println("//目录存在");
                }
                File newName = new File(copyDirectory + files[i].getName());
                if (files[i].renameTo(newName)) {
//                    System.out.println("File is moved successful!");//输出移动成功
                    log.info("File is moved successful!");
                } else {
//                    System.out.println("File is failed to move !");//输出移动失败
                    log.info("File is failed to move !");
                }
            }
        }
    }

    public static String getName(FTPClient ftp,String path) {
        String s = "";
        try {
            ftp.changeWorkingDirectory(new String(path.getBytes("GBK"),"ISO-8859-1"));
            ftp.enterLocalPassiveMode();
            //获取文件数组
            FTPFile[] files = ftp.listFiles();
            //回到原有工作目录
            ftp.changeWorkingDirectory(new String(path.getBytes("GBK"),"ISO-8859-1"));

            for (int i = 0; i < files.length; i++) {
                if (i == files.length-1) {
                    //转存目录
                    s = files[i].getName();
                }
            }
        } catch (Exception e) {
            e.printStackTrace();
            log.error("查询文件失败");
        }
        return s;
    }

    public static Map<String,String> getNames(FTPClient ftp,String path) {
        Map<String,String>  resMap = new HashMap<String,String>();
        try {
            ftp.changeWorkingDirectory(new String(path.getBytes("GBK"),"ISO-8859-1"));
            ftp.enterLocalPassiveMode();
            //获取文件数组
            FTPFile[] files = ftp.listFiles();
            //回到原有工作目录
            ftp.changeWorkingDirectory(new String(path.getBytes("GBK"),"ISO-8859-1"));

            for (int i = 0; i < files.length; i++) {
                resMap.put( Integer.toString(i) , files[i].getName() );
            }
        } catch (Exception e) {
            e.printStackTrace();
            log.error("查询文件失败");
        }
        return resMap;
    }

}
